import requests
import json
import sys


def test_stock_api(base_url="http://localhost:5000", auth_key="stockdata123"):
    """
    测试股票信息 HTTP 服务

    参数:
        base_url: 服务器基础 URL
        auth_key: 认证密钥
        stock_code: 要查询的股票代码
    """

    # 测试带参数的文本格式接口
    print("\n" + "=" * 50)
    print(f"测试股票 {stock_code} 的文本格式接口")
    print("=" * 50)
    try:
        # 将认证密钥从header移到URL查询参数中
        text_url = f"{base_url}/api/brentoil/text?fwkey={auth_key}"
        print(f"发送请求到: {text_url}")
        response = requests.get(text_url)

        # 检查响应状态
        print(f"响应状态码: {response.status_code}")
        if response.status_code == 200:
            print("\n接收到的文本响应:")
            print("-" * 50)
            print(response.text)
            print("-" * 50)
        else:
            print(f"错误: {response.text}")
    except Exception as e:
        print(f"文本接口请求发生错误: {str(e)}")


if __name__ == "__main__":
    # 允许从命令行指定参数
    # base_url = sys.argv[1] if len(sys.argv) > 1 else "http://localhost:35000"
    base_url = sys.argv[1] if len(sys.argv) > 1 else "http://14.103.252.171:35000"
    # auth_key = sys.argv[2] if len(sys.argv) > 2 else "stockdata123"
    auth_key = sys.argv[2] if len(sys.argv) > 2 else "MmSp5zH2Attcasj7UT7C"

    print(f"使用服务器 URL: {base_url}")
    print(f"使用认证密钥: {auth_key}")

    test_stock_api(base_url, auth_key)